package chatBot.service;

public interface TokenConsumptionService {

    /**
     * 计算总的token消耗
     *
     * @param promptTokens 提问的token数量
     * @param contextTokens 上下文的token数量
     * @param responseTokens 估算响应的token数量
     * @return 总的token消耗
     */
    int calculateTotalTokenConsumption(int promptTokens, int contextTokens, int responseTokens);

    /**
     * 计算字符串的token数量
     *
     * @param text 字符串内容
     * @return 计算出的token数量
     */
    int calculateTokenCount(String text);

    /**
     * 估算最大响应token数量
     *
     * @param promptTokens 提问的token数量
     * @param contextTokens 上下文的token数量
     * @return 最大响应的token数量
     */
    int calculateMaxResponseTokens(int promptTokens, int contextTokens);
}
